from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `tb_system_user` ADD `last_login_type` SMALLINT COMMENT '最后登录类型';"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `tb_system_user` DROP COLUMN `last_login_type`;"""
